Multi-Platform Comparison + Week 4 Review

import anthropic
import json

# Re-use the same helper functions from prompt_manager_v2.py
PROMPTS_FILE = "prompts.json"
PLATFORMS    = ["Kling", "Runway", "Veo"]

# Anthropic client — reads ANTHROPIC_API_KEY from environment automatically
client = anthropic.Anthropic()

def load_prompts():
    try:
        with open(PROMPTS_FILE, "r") as f:
            return json.load(f)
    except FileNotFoundError:
        return []

def save_prompts(prompts):
    with open(PROMPTS_FILE, "w") as f:
        json.dump(prompts, f, indent=2)

def generate_prompt(shot, platform):
    """Call Claude to generate a prompt tuned for the given platform."""
    message = client.messages.create(
        model="claude-sonnet-4-6",
        max_tokens=200,
        # system sets Claude's role — it stays in effect for the whole conversation
        system="You are an expert AI video prompt engineer.",
        messages=[{
            "role": "user",
            # f-string lets us inject shot and platform into the instruction
            "content": f"Write a {platform} AI video prompt for: {shot}. Match {platform}'s strengths. Under 75 words. Only the prompt."
        }]
    )
    # The response is a Message object; .content is a list of blocks;
    # [0].text gets the text from the first (and only) block
    return message.content[0].text

# Get the shot description from the user
shot = input("\nDescribe your shot: ").strip()

print("\nGenerating prompts — this may take a few seconds...\n")

# results will hold platform → generated prompt text (or error message)
results = {}

for platform in PLATFORMS:
    print(f"  Generating for {platform}...")
    try:
        # Call the API — this might take 1-3 seconds per platform
        results[platform] = generate_prompt(shot, platform)
    except Exception as e:
        # Catch ANY exception from the API call.
        # Store an error placeholder so display code can check for it.
        # The real error is available in 'e' if you need to debug.
        results[platform] = f"(Error: {e})"
        print(f"  Warning: {platform} failed — {e}")

# ── Display comparison ────────────────────────────────────────────────
print(f"\n{'=' * 50}")
print(f"  SHOT: {shot}")
print(f"{'=' * 50}\n")

# Loop through results dict — same order as PLATFORMS list
# because Python dicts preserve insertion order (Python 3.7+)
for platform, prompt in results.items():
    print(f"--- {platform} ---")
    print(f"{prompt}\n")

# ── Save options ──────────────────────────────────────────────────────
print("Save options: 'all', 'none', or platform names like 'Kling, Veo'")
choice = input("\nSave: ").strip().lower()

if choice == "all":
    # Save every result that isn't an error
    prompts = load_prompts()
    for platform, prompt in results.items():
        if not prompt.startswith("(Error"):
            prompts.append({"platform": platform, "shot": shot, "prompt": prompt})
    save_prompts(prompts)
    print("\n  All results saved to prompts.json\n")

elif choice == "none":
    # User just wanted to compare — nothing saved
    print("\n  Nothing saved.\n")

else:
    # Parse "kling, veo" → ["kling", "veo"]
    # .split(",") splits on commas; .strip() removes spaces around each name
    chosen = [c.strip().lower() for c in choice.split(",")]

    prompts     = load_prompts()
    saved_count = 0

    for platform, prompt in results.items():
        # Check: was this platform in the user's chosen list?
        # Was it a successful result (not an error)?
        if platform.lower() in chosen and not prompt.startswith("(Error"):
            prompts.append({"platform": platform, "shot": shot, "prompt": prompt})
            saved_count += 1

    save_prompts(prompts)
    print(f"\n  Saved {saved_count} prompt(s) to prompts.json\n")